import unittest
# 导入测试类
from test01 import TestAdd
from test02 import TestAvg

# 创建测试套件
suite = unittest.TestSuite()
# 添加测试用例
suite.addTest(TestAdd('test_add01'))
suite.addTest(TestAdd('test_add02'))
suite.addTest(TestAdd('test_add03'))
# 将测试类中所有以test开头的方法到测试套件中
suite.addTest(unittest.makeSuite(TestAvg))

if __name__ == '__main__':
    # 创建测试套件运行器
    runner = unittest.TextTestRunner()
    # 执行测试套件
    runner.run(suite)